package com.lee.mq.mqserver.core;

/**
 * Created with IntelliJ IDEA.
 * Description:
 * User: lishuo
 * Date: 2023-10-19
 * Time: 16:33
 */

import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.core.type.TypeReference;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.lee.mq.common.ConsumerEnv;
import com.lee.mq.common.MqException;

import javax.xml.ws.soap.Addressing;
import java.util.*;
import java.util.concurrent.atomic.AtomicInteger;

/**
 * 表示存储消息的队列的类
 */
public class MSGQueue {
    // name 表示队列的身份标识
    private String name;
    // durable 表示队列是否持久化（true 表示持久化保存；false 表示不持久化）
    private boolean durable = false;

    // autoDelete 表示自动删除。如果当前交换机不再使用就会被自动删除。
    // 该属性在 RabbitMQ 中存在，这里作为拓展属性（自行实现）
    private boolean autoDelete = false;

    // exclusive 表示队列的独占功能。（true 表示该队列只能被一个消费者使用；false 表示都能使用）
    // 该属性在 RabbitMQ 中存在，这里作为拓展属性（自行实现）
    private boolean exclusive = false;

    // arguments 表示创建队列时的一些额外的参数。
    // 为了实现 arguments 在数据库中的存取, 就需要把 Map 转成 json 格式的字符串，并对JSON字符串进行解析.
    // 因此需要调整一下 argument 的 get 和 set 方法.
    private Map<String,Object> arguments = new HashMap<>();

    // 这里添加一个队列的订阅者列表
    private List<ConsumerEnv> consumerEvnList = new ArrayList<>();
    // 添加订阅者方法
    public void addConsumerEvn(ConsumerEnv consumerEnv) {
        consumerEvnList.add(consumerEnv);
    }
    // atomicInteger 用于轮播策略
    private AtomicInteger atomicInteger = new AtomicInteger(0);
    // 使用轮播方式获取消费环境，便于后续进行具体的消息处理
    public ConsumerEnv chooseConsumerEvn() {
        if (consumerEvnList.size() == 0) {
            // 当前对列不存在订阅者
            return null;
        }
        // 计算轮播下标
        int index = atomicInteger.get()%consumerEvnList.size();
        atomicInteger.getAndIncrement();
        return consumerEvnList.get(index);
    }
    // 暂不考虑订阅者删除

    public String getName() {
        return name;
    }

    public void setName(String name) {
        this.name = name;
    }

    public boolean isDurable() {
        return durable;
    }

    public void setDurable(boolean durable) {
        this.durable = durable;
    }

    public boolean isAutoDelete() {
        return autoDelete;
    }

    public void setAutoDelete(boolean autoDelete) {
        this.autoDelete = autoDelete;
    }

    public boolean isExclusive() {
        return exclusive;
    }

    public void setExclusive(boolean exclusive) {
        this.exclusive = exclusive;
    }

    // 写入数据库时
    public String getArguments() {
        ObjectMapper objectMapper = new ObjectMapper();
        try {
            return objectMapper.writeValueAsString(arguments);
        } catch (JsonProcessingException e) {
            e.printStackTrace();
        }
        return "{}";
    }

    // 读取数据库时
    public void setArguments(String argumentsJson) {
        ObjectMapper objectMapper = new ObjectMapper();
        try {
            this.arguments = objectMapper.readValue(argumentsJson,
                    new TypeReference<Map<String, Object>>() {});
        } catch (JsonProcessingException e) {
            e.printStackTrace();
        }
    }

    // 提供一组 getter setter , 用来更方便的获取/设置这里的键值对.
    // 这一组在 java 代码内部使用 (比如测试的时候)
    public Object getArguments(String key) {
        return arguments.get(key);
    }

    public void setArguments(String key,Object value) {
        arguments.put(key,value);
    }

    public void setArguments(Map<String,Object> arguments) {
        this.arguments = arguments;
    }

}
